home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / share / pyshared / DistUpgrade / DistUpgradeView.py < prev    next >
Encoding:
Python Source  |  2009-04-27  |  11.7 KB  |  325 lines
  1. # DistUpgradeView.py 
  2. #  
  3. #  Copyright (c) 2004,2005 Canonical
  4. #  
  5. #  Author: Michael Vogt <michael.vogt@ubuntu.com>
  6. # 
  7. #  This program is free software; you can redistribute it and/or 
  8. #  modify it under the terms of the GNU General Public License as 
  9. #  published by the Free Software Foundation; either version 2 of the
  10. #  License, or (at your option) any later version.
  11. # 
  12. #  This program is distributed in the hope that it will be useful,
  13. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15. #  GNU General Public License for more details.
  16. # 
  17. #  You should have received a copy of the GNU General Public License
  18. #  along with this program; if not, write to the Free Software
  19. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  20. #  USA
  21.  
  22. from DistUpgradeGettext import gettext as _
  23. from DistUpgradeGettext import ngettext
  24. import subprocess
  25. from subprocess import Popen, PIPE
  26. import apt
  27. import os
  28. import apt_pkg 
  29. import signal
  30. import glob
  31.  
  32. from DistUpgradeAufs import doAufsChroot, doAufsChrootRsync
  33. from DistUpgradeApport import *
  34.  
  35.  
  36. def FuzzyTimeToStr(sec):
  37.   " return the time a bit fuzzy (no seconds if time > 60 secs "
  38.   #print "FuzzyTimeToStr: ", sec
  39.   sec = int(sec)
  40.  
  41.   days = sec/(60*60*24)
  42.   hours = sec/(60*60) % 24
  43.   minutes = (sec/60) % 60
  44.   seconds = sec % 60
  45.   # 0 seonds remaining looks wrong and its "fuzzy" anyway
  46.   if seconds == 0:
  47.     seconds = 1
  48.  
  49.   # string map to make the re-ordering possible
  50.   map = { "str_days" : "",
  51.           "str_hours" : "",
  52.           "str_minutes" : "",
  53.           "str_seconds" : ""
  54.         }
  55.   
  56.   # get the fragments, this is not ideal i18n wise, but its
  57.   # difficult to do it differently
  58.   if days > 0:
  59.     map["str_days"] = ngettext("%li day","%li days", days) % days
  60.   if hours > 0:
  61.     map["str_hours"] = ngettext("%li hour","%li hours", hours) % hours
  62.   if minutes > 0:
  63.     map["str_minutes"] = ngettext("%li minute","%li minutes", minutes) % minutes
  64.   map["str_seconds"] = ngettext("%li second","%li seconds", seconds) % seconds
  65.  
  66.   # now assemble the string
  67.   if days > 0:
  68.     # TRANSLATORS: you can alter the ordering of the remaining time
  69.     # information here if you shuffle %(str_days)s %(str_hours)s %(str_minutes)s
  70.     # around. Make sure to keep all '$(str_*)s' in the translated string
  71.     # and do NOT change anything appart from the ordering.
  72.     #
  73.     # %(str_hours)s will be either "1 hour" or "2 hours" depending on the
  74.     # plural form
  75.     # 
  76.     # Note: most western languages will not need to change this
  77.     return _("%(str_days)s %(str_hours)s") % map
  78.   # display no minutes for time > 3h, see LP: #144455
  79.   elif hours > 3:
  80.     return map["str_hours"]
  81.   # when we are near the end, become more precise again
  82.   elif hours > 0:
  83.     # TRANSLATORS: you can alter the ordering of the remaining time
  84.     # information here if you shuffle %(str_hours)s %(str_minutes)s
  85.     # around. Make sure to keep all '$(str_*)s' in the translated string
  86.     # and do NOT change anything appart from the ordering.
  87.     #
  88.     # %(str_hours)s will be either "1 hour" or "2 hours" depending on the
  89.     # plural form
  90.     # 
  91.     # Note: most western languages will not need to change this
  92.     return _("%(str_hours)s %(str_minutes)s") % map
  93.   elif minutes > 0:
  94.     return map["str_minutes"]
  95.   return map["str_seconds"]
  96.  
  97.  
  98. class FetchProgress(apt.progress.FetchProgress):
  99.   def __init__(self):
  100.     #print "init FetchProgress in DistUpgradeView"
  101.     apt.progress.FetchProgress.__init__(self)
  102.     self.est_speed = 0
  103.   def pulse(self):
  104.     apt.progress.FetchProgress.pulse(self)
  105.     if self.currentCPS > self.est_speed:
  106.       self.est_speed = (self.est_speed+self.currentCPS)/2.0
  107.     return True
  108.   def estimatedDownloadTime(self, requiredDownload):
  109.     """ get the estimated download time """
  110.     if self.est_speed == 0:
  111.       timeModem = requiredDownload/(56*1024/8)  # 56 kbit 
  112.       timeDSL = requiredDownload/(1024*1024/8)  # 1Mbit = 1024 kbit
  113.       s= _("This download will take about %s with a 1Mbit DSL connection "
  114.            "and about %s with a 56k modem.") % (FuzzyTimeToStr(timeDSL), FuzzyTimeToStr(timeModem))
  115.       return s
  116.     # if we have a estimated speed, use it
  117.     s = _("This download will take about %s with your connection. ") % FuzzyTimeToStr(requiredDownload/self.est_speed)
  118.     return s
  119.     
  120.  
  121.  
  122. class InstallProgress(apt.progress.InstallProgress):
  123.   """ Base class for InstallProgress that supports some fancy
  124.       stuff like apport integration
  125.   """
  126.   def __init__(self):
  127.     apt.progress.InstallProgress.__init__(self)
  128.     self.master_fd = None
  129.  
  130.   def run(self, pm):
  131.     pid = self.fork()
  132.     if pid == 0:
  133.       # check if we need to setup/enable the aufs chroot stuff
  134.       if "RELEASE_UPGRADE_USE_AUFS_CHROOT" in os.environ:
  135.         if not doAufsChroot(os.environ["RELEASE_UPGRADE_AUFS_RWDIR"],
  136.                             os.environ["RELEASE_UPGRADE_USE_AUFS_CHROOT"]):
  137.           print "ERROR: failed to setup aufs chroot overlay"
  138.           os._exit(1)
  139.       # child, ignore sigpipe, there are broken scripts out there
  140.       # like etckeeper (LP: #283642)
  141.       signal.signal(signal.SIGPIPE,signal.SIG_IGN) 
  142.       res = pm.DoInstall(self.writefd)
  143.       os._exit(res)
  144.     self.child_pid = pid
  145.     res = self.waitChild()
  146.     # check if we want to sync the changes back, *only* do that
  147.     # if res is positive
  148.     if (res == 0 and
  149.         "RELEASE_UPGRADE_RSYNC_AUFS_CHROOT" in os.environ):
  150.       logging.info("doing rsync commit of the update")
  151.       if not doAufsChrootRsync(os.environ["RELEASE_UPGRADE_USE_AUFS_CHROOT"]):
  152.         logging.error("FATAL ERROR: doAufsChrootRsync() returned FALSE")
  153.         return False
  154.     return res
  155.   
  156.   def error(self, pkg, errormsg):
  157.     " install error from a package "
  158.     apt.progress.InstallProgress.error(self, pkg, errormsg)
  159.     logging.error("got an error from dpkg for pkg: '%s': '%s'" % (pkg, errormsg))
  160.     if "/" in pkg:
  161.       pkg = os.path.basename(pkg)
  162.     if "_" in pkg:
  163.       pkg = pkg.split("_")[0]
  164.     # now run apport
  165.     apport_pkgfailure(pkg, errormsg)
  166.  
  167. class DumbTerminal(object):
  168.     def call(self, cmd, hidden=False):
  169.         " expects a command in the subprocess style (as a list) "
  170.         import subprocess
  171.         subprocess.call(cmd)
  172.  
  173.  
  174. (STEP_PREPARE,
  175.  STEP_MODIFY_SOURCES,
  176.  STEP_FETCH,
  177.  STEP_INSTALL,
  178.  STEP_CLEANUP,
  179.  STEP_REBOOT,
  180.  STEP_N) = range(1,8)
  181.  
  182. ( _("Preparing to upgrade"),
  183.   _("Getting new software channels"),
  184.   _("Getting new packages"),
  185.   _("Installing the upgrades"),
  186.   _("Cleaning up"),
  187. )
  188.  
  189. class DistUpgradeView(object):
  190.     " abstraction for the upgrade view "
  191.     def __init__(self):
  192.         pass
  193.     def getOpCacheProgress(self):
  194.         " return a OpProgress() subclass for the given graphic"
  195.         return apt.progress.OpProgress()
  196.     def getFetchProgress(self):
  197.         " return a fetch progress object "
  198.         return FetchProgress()
  199.     def getInstallProgress(self, cache=None):
  200.         " return a install progress object "
  201.         return InstallProgress()
  202.     def getTerminal(self):
  203.         return DumbTerminal()
  204.     def updateStatus(self, msg):
  205.         """ update the current status of the distUpgrade based
  206.             on the current view
  207.         """
  208.         pass
  209.     def abort(self):
  210.         """ provide a visual feedback that the upgrade was aborted """
  211.         pass
  212.     def setStep(self, step):
  213.         """ we have 6 steps current for a upgrade:
  214.         1. Analyzing the system
  215.         2. Updating repository information
  216.         3. fetch packages
  217.         3. Performing the upgrade
  218.         4. Post upgrade stuff
  219.         5. Complete
  220.         """
  221.         pass
  222.     def hideStep(self, step):
  223.         " hide a certain step from the GUI "
  224.         pass
  225.     def showStep(self, step):
  226.         " show a certain step from the GUI "
  227.         pass
  228.     def confirmChanges(self, summary, changes, downloadSize,
  229.                        actions=None, removal_bold=True):
  230.         """ display the list of changed packages (apt.Package) and
  231.             return if the user confirms them
  232.         """
  233.         self.confirmChangesMessage = ""
  234.         self.toInstall = []
  235.         self.toUpgrade = []
  236.         self.toRemove = []
  237.         self.toDowngrade = []
  238.         for pkg in changes:
  239.             if pkg.markedInstall: self.toInstall.append(pkg.name)
  240.             elif pkg.markedUpgrade: self.toUpgrade.append(pkg.name)
  241.             elif pkg.markedDelete: self.toRemove.append(pkg.name)
  242.             elif pkg.markedDowngrade: self.toDowngrade.append(pkg.name)
  243.         # sort it
  244.         self.toInstall.sort()
  245.         self.toUpgrade.sort()
  246.         self.toRemove.sort()
  247.         self.toDowngrade.sort()
  248.         # no re-installs 
  249.         assert(len(self.toInstall)+len(self.toUpgrade)+len(self.toRemove)+len(self.toDowngrade) == len(changes))
  250.         # now build the message (the same for all frontends)
  251.         msg = "\n"
  252.         pkgs_remove = len(self.toRemove)
  253.         pkgs_inst = len(self.toInstall)
  254.         pkgs_upgrade = len(self.toUpgrade)
  255.         # FIXME: show detailed packages
  256.         if pkgs_remove > 0:
  257.           # FIXME: make those two separate lines to make it clear
  258.           #        that the "%" applies to the result of ngettext
  259.           msg += ngettext("%d package is going to be removed.",
  260.                           "%d packages are going to be removed.",
  261.                           pkgs_remove) % pkgs_remove
  262.           msg += " "
  263.         if pkgs_inst > 0:
  264.           msg += ngettext("%d new package is going to be "
  265.                           "installed.",
  266.                           "%d new packages are going to be "
  267.                           "installed.",pkgs_inst) % pkgs_inst
  268.           msg += " "
  269.         if pkgs_upgrade > 0:
  270.           msg += ngettext("%d package is going to be upgraded.",
  271.                           "%d packages are going to be upgraded.",
  272.                           pkgs_upgrade) % pkgs_upgrade
  273.           msg +=" "
  274.         if downloadSize > 0:
  275.           msg += _("\n\nYou have to download a total of %s. ") %\
  276.               apt_pkg.SizeToStr(downloadSize)
  277.           msg += self.getFetchProgress().estimatedDownloadTime(downloadSize)
  278.         if (pkgs_upgrade + pkgs_inst + pkgs_remove) > 100:
  279.           msg += "\n\n%s" % _( "Fetching and installing the upgrade "
  280.                                "can take several hours. Once the download "
  281.                                "has finished, the process cannot be cancelled.")
  282.         # Show an error if no actions are planned
  283.         if (pkgs_upgrade + pkgs_inst + pkgs_remove) < 1:
  284.           # FIXME: this should go into DistUpgradeController
  285.           summary = _("Your system is up-to-date")
  286.           msg = _("There are no upgrades available for your system. "
  287.                   "The upgrade will now be canceled.")
  288.           self.error(summary, msg)
  289.           return False
  290.         # set the message
  291.         self.confirmChangesMessage = msg
  292.         return True
  293.  
  294.     def askYesNoQuestion(self, summary, msg, default='No'):
  295.         " ask a Yes/No question and return True on 'Yes' "
  296.         pass
  297.     def confirmRestart(self):
  298.         " generic ask about the restart, can be overridden "
  299.         summary = _("Reboot required")
  300.         msg =  _("The upgrade is finished and "
  301.                  "a reboot is required. "
  302.                  "Do you want to do this "
  303.                  "now?")
  304.         return self.askYesNoQuestion(summary, msg)
  305.     def error(self, summary, msg, extended_msg=None):
  306.         " display a error "
  307.         pass
  308.     def information(self, summary, msg, extended_msg=None):
  309.         " display a information msg"
  310.         pass
  311.     def processEvents(self):
  312.         """ process gui events (to keep the gui alive during a long
  313.             computation """
  314.         pass
  315.     def showDemotions(self, summary, msg, demotions):
  316.       """
  317.       show demoted packages to the user, default implementation
  318.       is to just show a information dialog
  319.       """
  320.       self.information(summary, msg, "\n".join(demotions))
  321.  
  322. if __name__ == "__main__":
  323.   fp = FetchProgress()
  324.   fp.pulse()
  325.